#pragma once
#include <iostream>
#include <string>
#include <vector>
#include <atomic>
#include <fstream>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <sys/time.h>
#include <boost/algorithm/string.hpp>

namespace ns_util
{
    //封装时间工具类
    class TimeUtil
    {
    public:
        //该函数用来获取系统的时间戳
        static std::string GetTimeStamp()
        {
            struct timeval _time;
            //系统调用接口gettimeofday，第一个参数为输出型参数，第二个参数是时区，设置为nullptr即可
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec);
        }

        //获得毫秒时间戳
        static std::string GetTimeMs()
        {
            struct timeval _time;
            //系统调用接口gettimeofday，第一个参数为输出型参数，第二个参数是时区，设置为nullptr即可
            gettimeofday(&_time, nullptr);
            return std::to_string(_time.tv_sec * 1000 + _time.tv_usec / 1000);
        }
    };

    //需要使用的全局路径
    const std::string temp_path = "./temp/";

    //封装对路径进行操作的工具类
    class PathUtil
    {
    public:
    
        /*编译时需要的临时文件*/

        //构建文件+后缀的的完整文件名
        static std::string AddSuffix(const std::string &file_name, const std::string &suffix)
        {
            std::string path_name = temp_path;
            path_name += file_name;
            path_name += suffix;
            return path_name;
        }

        // 构建源文件路径+后缀的完整文件名
        static std::string Src(const std::string &file_name)
        {
            return AddSuffix(file_name, ".cpp");
        }


        // 构建可执行程序的完整路径+后缀名
        static std::string Exe(const std::string &file_name)
        {
            return AddSuffix(file_name, ".exe");
        }
        
        // 构建编译时报错的文件
        static std::string CompilerError(const std::string &file_name)
        {
            return AddSuffix(file_name, ".compile_error");
        }

        /*运行时需要的临时文件*/

        // 构建该程序对应的标准输入完整的路径+后缀名
        static std::string Stdin(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdin");
        }

        // 构建该程序对应的标准输出完整的路径+后缀名
        static std::string Stdout(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdout");
        }
        
        // 构建该程序对应的标准错误完整的路径+后缀名
        static std::string Stderr(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stderr");
        }
        
    };

    //封装对文件操作的类
    class FileUtil
    {
    public:
        //该函数判断文件是否存在
        static bool IsFileExists(const std::string &path_name)
        {
            //stat是系统调用接口，检测文件的相关属性，传入到第二个参数中
            struct stat st;
            if (stat(path_name.c_str(), &st) == 0)
            {
                //获取属性成功，说明文件已经存在
                return true;
            }
            return false; //文件不存在
        }

        // 该函数用来生成唯一的文件名 
        // 编译服务随时可能被多个人请求，必须保证传递上来的code，形成源文件名称的时候，要具有唯一性，要不然多个用户之间会互相影响
        static std::string UniqFileName()
        {
            // 毫秒级时间戳 + 原子性递增唯一值，来保证唯一性
            //C++11原子库
            static std::atomic_uint id(0);
            id++;
            std::string ms = TimeUtil::GetTimeMs();
            std::string uniq_id = std::to_string(id);
            return ms + "_" + uniq_id;
        }

        //向目标文件中写内容
        static bool WriteFile(const std::string &target, const std::string &content)
        {
            std::ofstream out(target);
            if (!out.is_open())
            {
                return false;
            }
            //参数：要写的内容，多少字节
            out.write(content.c_str(), content.size());
            out.close();
            return true;
        }

        //读取目标文件中的内容
        static bool ReadFile(const std::string &target, std::string *content, bool keep = false)
        {
            (*content).clear();
            std::ifstream in(target);
            if (!in.is_open())
            {
                return false;
            }
            std::string line;
            // getline参数：从哪里读，读到哪个地方
            while (std::getline(in, line))
            {
                (*content) += line;
                // getline:不保存行分割符,但是有些时候需要保留\n,
                (*content) += (keep ? "\n" : "");
            }
            in.close();
            return true;
        }
    };

    //字符串工具类
    class StringUtil
    {
    public:
        /*************************************
         * str: 输入型，目标要切分的字符串
         * target: 输出型，保存切分完毕的结果
         * sep: 指定的分割符
         * **********************************/
        static void SplitString(const std::string &str, std::vector<std::string> *target, const std::string &sep)
        {
            //使用boost库的split函数用来切割字符串 
            boost::split((*target), str, boost::is_any_of(sep), boost::algorithm::token_compress_on);
        }
    };
}


